甲方|详情_基于Go语言实现OSS文件上传（local）

篇首语：本文由编程笔记#小编为大家整理，主要介绍了基于Go语言实现OSS文件上传（local）相关的知识，希望对你有一定的参考价值。

目录

• Config&＃xff1a;
• • server/config/oss.go
  • server/config/config.go
  • server/config.yaml
  
  
• Model&＃xff1a;
• • server/model/example/exa_file_upload_and_download.go
  
  
• Utils&＃xff1a;
• • server/utils/upload.go
  • server/utils/local.go
  
  
• Service&＃xff1a;
• • server/service/example/exa_file_upload_download.go
  
  
• Controller&＃xff1a;
• server/api/exa_file_upload_download.go
• 接口测试&＃xff1a;

server/config/oss.go

package config
type Local struct
Path string &＃96;mapstructure:"path" json:"path" yaml:"path" &＃96;

type Qiniu struct
Zone string &＃96;mapstructure:"zone" json:"zone" yaml:"zone"&＃96;
Bucket string &＃96;mapstructure:"bucket" json:"bucket" yaml:"bucket"&＃96;
ImgPath string &＃96;mapstructure:"img-path" json:"imgPath" yaml:"img-path"&＃96;
UseHTTPS bool &＃96;mapstructure:"use-https" json:"useHttps" yaml:"use-https"&＃96;
AccessKey string &＃96;mapstructure:"access-key" json:"accessKey" yaml:"access-key"&＃96;
SecretKey string &＃96;mapstructure:"secret-key" json:"secretKey" yaml:"secret-key"&＃96;
UseCdnDomains bool &＃96;mapstructure:"use-cdn-domains" json:"useCdnDomains" yaml:"use-cdn-domains"&＃96;

type Minio struct
Id string &＃96;mapstructure:"id" json:"id" yaml:"id"&＃96;
Path string &＃96;mapstructure:"path" json:"path" yaml:"path"&＃96;
Token string &＃96;mapstructure:"token" json:"token" yaml:"token"&＃96;
Bucket string &＃96;mapstructure:"bucket" json:"bucket" yaml:"bucket"&＃96;
UseSsl bool &＃96;mapstructure:"use-ssl" json:"useSsl" yaml:"use-ssl"&＃96;
Secret string &＃96;mapstructure:"secret" json:"secret" yaml:"secret"&＃96;
Endpoint string &＃96;mapstructure:"endpoint" json:"endpoint" yaml:"endpoint"&＃96;

type Aliyun struct
Path string &＃96;mapstructure:"path" json:"path" yaml:"path"&＃96;
Bucket string &＃96;mapstructure:"bucket" json:"bucket" yaml:"bucket"&＃96;
ACLType string &＃96;mapstructure:"acl-type" json:"aclType" yaml:"acl-type"&＃96;
Endpoint string &＃96;mapstructure:"endpoint" json:"endpoint" yaml:"endpoint"&＃96;
AccessKeyID string &＃96;mapstructure:"access-key-id" json:"accessKeyId" yaml:"access-key-id"&＃96;
SecretAccessKey string &＃96;mapstructure:"secret-access-key" json:"secretAccessKey" yaml:"secret-access-key"&＃96;
StorageClassType string &＃96;mapstructure:"storage-class-type" json:"storageClassType" yaml:"storage-class-type"&＃96;



server/config/config.go

type Server struct
JWT JWT &＃96;mapstructure:"jwt" json:"jwt" yaml:"jwt"&＃96;
Zap Zap &＃96;mapstructure:"zap" json:"zap" yaml:"zap"&＃96;
Redis Redis &＃96;mapstructure:"redis" json:"redis" yaml:"redis"&＃96;
//Email Email &＃96;mapstructure:"email" json:"email" yaml:"email"&＃96;
//Casbin Casbin &＃96;mapstructure:"casbin" json:"casbin" yaml:"casbin"&＃96;
System System &＃96;mapstructure:"system" json:"system" yaml:"system"&＃96;
Captcha Captcha &＃96;mapstructure:"captcha" json:"captcha" yaml:"captcha"&＃96;
// auto
AutoCode Autocode &＃96;mapstructure:"autoCode" json:"autoCode" yaml:"autoCode"&＃96;
// gorm
mysql Mysql &＃96;mapstructure:"mysql" json:"mysql" yaml:"mysql"&＃96;
//Pgsql Pgsql &＃96;mapstructure:"pgsql" json:"pgsql" yaml:"pgsql"&＃96;
//DBList []DB &＃96;mapstructure:"db-list" json:"db-list" yaml:"db-list"&＃96;
// oss
Local Local &＃96;mapstructure:"local" json:"local" yaml:"local"&＃96;
//Qiniu Qiniu &＃96;mapstructure:"qiniu" json:"qiniu" yaml:"qiniu"&＃96;
//AliyunOSS AliyunOSS &＃96;mapstructure:"aliyun-oss" json:"aliyunOSS" yaml:"aliyun-oss"&＃96;
//HuaWeiObs HuaWeiObs &＃96;mapstructure:"hua-wei-obs" json:"huaWeiObs" yaml:"hua-wei-obs"&＃96;
//TencentCOS TencentCOS &＃96;mapstructure:"tencent-cos" json:"tencentCOS" yaml:"tencent-cos"&＃96;
AwsS3 AwsS3 &＃96;mapstructure:"aws-s3" json:"awsS3" yaml:"aws-s3"&＃96;
//
//Excel Excel &＃96;mapstructure:"excel" json:"excel" yaml:"excel"&＃96;
//Timer Timer &＃96;mapstructure:"timer" json:"timer" yaml:"timer"&＃96;
//
// 跨域配置
Cors CORS &＃96;mapstructure:"cors" json:"cors" yaml:"cors"&＃96;



server/config.yaml

# local configuration
local:
path: &＃39;uploads/file&＃39;
# qiniu configuration (请自行七牛申请对应的 公钥 私钥 bucket 和 域名地址)
qiniu:
zone: &＃39;ZoneHuadong&＃39;
bucket: &＃39;qm-plus-img&＃39;
img-path: &＃39;http://qmplusimg.henrongyi.top&＃39;
use-https: false
access-key: &＃39;25j8dYBZ2wuiy0yhwShytjZDTX662b8xiFguwxzZ&＃39;
secret-key: &＃39;pgdbqEsf7ooZh7W3xokP833h3dZ_VecFXPDeG5JY&＃39;
use-cdn-domains: false
# minio configuration
minio:
id: &＃39;minio&＃39;
path: &＃39;http://localhost:9000&＃39;
token: &＃39;&＃39;
bucket: &＃39;public&＃39;
use-ssl: false
secret: &＃39;minio1234&＃39;
endpoint: &＃39;localhost:9000&＃39;
# aliyun configuration
aliyun:
path : &＃39;https://xxx.oss-cn-beijing.aliyuncs.com&＃39;
bucket : &＃39;xxx&＃39;
acl-type : &＃39;&＃39;
endpoint : &＃39;https://oss-cn-beijing.aliyuncs.com&＃39;
access-key-id : &＃39;xxxxxxx&＃39;
secret-access-key : &＃39;xxxxxxx&＃39;
storage-class-type : &＃39;Standard&＃39; # 存储桶存储类型

Model&＃xff1a;

server/model/example/exa_file_upload_and_download.go

package example
import (
"Project/global"
)
type ExaFileUploadAndDownload struct
global.KBB_MODEL
Name string &＃96;json:"name" gorm:"comment:文件名"&＃96; // 文件名
Url string &＃96;json:"url" gorm:"comment:文件地址"&＃96; // 文件地址
Tag string &＃96;json:"tag" gorm:"comment:文件标签"&＃96; // 文件标签
Key string &＃96;json:"key" gorm:"comment:编号"&＃96; // 编号

func (E ExaFileUploadAndDownload) TableName() string
return "exa_file_upload_and_download"


Utils&＃xff1a;

server/utils/upload.go

package upload
import (
"mime/multipart"
"Project/global"
)
// OSS 对象存储接口
// Author [SliverHorn](https://github.com/SliverHorn)
// Author [ccfish86](https://github.com/ccfish86)
type OSS interface
UploadFile(file *multipart.FileHeader) (string, string, error)
DeleteFile(key string) error

// NewOss OSS的实例化方法
// Author [SliverHorn](https://github.com/SliverHorn)
// Author [ccfish86](https://github.com/ccfish86)
func NewOss() OSS
switch global.KBB_CONFIG.System.OssType
case "local":
return &Local
case "aws-s3":
return &AwsS3
default:
return &Local




server/utils/local.go

package upload
import (
"errors"
"io"
"mime/multipart"
"os"
"path"
"strings"
"time"
"Project/global"
"Project/utils"
"go.uber.org/zap"
)
type Local struct
//&＃64;author: [piexlmax](https://github.com/piexlmax)
//&＃64;author: [ccfish86](https://github.com/ccfish86)
//&＃64;author: [SliverHorn](https://github.com/SliverHorn)
//&＃64;object: *Local
//&＃64;function: UploadFile
//&＃64;description: 上传文件
//&＃64;param: file *multipart.FileHeader
//&＃64;return: string, string, error
func (*Local) UploadFile(file *multipart.FileHeader) (string, string, error)
// 读取文件后缀
ext :&＃61; path.Ext(file.Filename)
// 读取文件名并加密
name :&＃61; strings.TrimSuffix(file.Filename, ext)
name &＃61; utils.MD5V([]byte(name))
// 拼接新文件名
filename :&＃61; name &＃43; "_" &＃43; time.Now().Format("20060102150405") &＃43; ext
// 尝试创建此路径
mkdirErr :&＃61; os.MkdirAll(global.KBB_CONFIG.Local.Path, os.ModePerm)
if mkdirErr !&＃61; nil
global.KBB_LOG.Error("function os.MkdirAll() Filed", zap.Any("err", mkdirErr.Error()))
return "", "", errors.New("function os.MkdirAll() Filed, err:" &＃43; mkdirErr.Error())

// 拼接路径和文件名
p :&＃61; global.KBB_CONFIG.Local.Path &＃43; "/" &＃43; filename
f, openError :&＃61; file.Open() // 读取文件
if openError !&＃61; nil
global.KBB_LOG.Error("function file.Open() Filed", zap.Any("err", openError.Error()))
return "", "", errors.New("function file.Open() Filed, err:" &＃43; openError.Error())

defer f.Close() // 创建文件 defer 关闭
out, createErr :&＃61; os.Create(p)
if createErr !&＃61; nil
global.KBB_LOG.Error("function os.Create() Filed", zap.Any("err", createErr.Error()))
return "", "", errors.New("function os.Create() Filed, err:" &＃43; createErr.Error())

defer out.Close() // 创建文件 defer 关闭
_, copyErr :&＃61; io.Copy(out, f) // 传输&＃xff08;拷贝&＃xff09;文件
if copyErr !&＃61; nil
global.KBB_LOG.Error("function io.Copy() Filed", zap.Any("err", copyErr.Error()))
return "", "", errors.New("function io.Copy() Filed, err:" &＃43; copyErr.Error())

return p, filename, nil

//&＃64;author: [piexlmax](https://github.com/piexlmax)
//&＃64;author: [ccfish86](https://github.com/ccfish86)
//&＃64;author: [SliverHorn](https://github.com/SliverHorn)
//&＃64;object: *Local
//&＃64;function: DeleteFile
//&＃64;description: 删除文件
//&＃64;param: key string
//&＃64;return: error
func (*Local) DeleteFile(key string) error
p :&＃61; global.KBB_CONFIG.Local.Path &＃43; "/" &＃43; key
if strings.Contains(p, global.KBB_CONFIG.Local.Path)
if err :&＃61; os.Remove(p); err !&＃61; nil
return errors.New("本地文件删除失败, err:" &＃43; err.Error())


return nil


Service&＃xff1a;

server/service/example/exa_file_upload_download.go

//&＃64;author: [piexlmax](https://github.com/piexlmax)
//&＃64;function: Upload
//&＃64;description: 创建文件上传记录
//&＃64;param: file model.ExaFileUploadAndDownload
//&＃64;return: error
func (e *FileUploadAndDownloadService) Upload(file example.ExaFileUploadAndDownload) error
return global.KBB_DB.Create(&file).Error

//&＃64;author: [piexlmax](https://github.com/piexlmax)
//&＃64;function: UploadFile
//&＃64;description: 根据配置文件判断是文件上传到本地或者七牛云
//&＃64;param: header *multipart.FileHeader, noSave string
//&＃64;return: err error, file model.ExaFileUploadAndDownload
func (e *FileUploadAndDownloadService) UploadFile(header *multipart.FileHeader, noSave string) (err error, file example.ExaFileUploadAndDownload)
oss :&＃61; upload.NewOss()
filePath, key, uploadErr :&＃61; oss.UploadFile(header)
if uploadErr !&＃61; nil
panic(err)

if noSave &＃61;&＃61; "0"
s :&＃61; strings.Split(header.Filename, ".")
f :&＃61; example.ExaFileUploadAndDownload
Url: filePath,
Name: header.Filename,
Tag: s[len(s)-1],
Key: key,

return e.Upload(f), f

return


Controller&＃xff1a;
server/api/exa_file_upload_download.go

// UploadFile &＃64;Tags ExaFileUploadAndDownload
// &＃64;Summary 上传文件示例
// &＃64;Security ApiKeyAuth
// &＃64;accept multipart/form-data
// &＃64;Produce application/json
// &＃64;Param file formData file true "上传文件示例"
// &＃64;Success 200 object response.Responsedata&＃61;exampleRes.ExaFileResponse,msg&＃61;string "上传文件示例,返回包括文件详情"
// &＃64;Router /fileUploadAndDownload/upload [post]
func (u *FileUploadAndDownloadApi) UploadFile(c *gin.Context)
var file example.ExaFileUploadAndDownload
noSave :&＃61; c.DefaultQuery("noSave", "0")
_, header, err :&＃61; c.Request.FormFile("file")
if err !&＃61; nil
global.KBB_LOG.Error("接收文件失败!", zap.Error(err))
utils.FailWithMessage("接收文件失败", c)
return

err, file &＃61; fileUploadAndDownloadService.UploadFile(header, noSave) // 文件上传后拿到文件路径
if err !&＃61; nil
global.KBB_LOG.Error("修改数据库链接失败!", zap.Error(err))
utils.FailWithMessage("修改数据库链接失败", c)
return

utils.OkWithDetailed(exampleRes.ExaFileResponseFile: file, "上传成功", c)


接口测试&＃xff1a;

由于资金有限&＃xff0c;甲方没有提供OSS服务器&＃xff0c;所以此接口&＃xff0c;只用于上传到local。
gin-vue-admin官方提供了qiniu、minio、aliyun等utils工具。

【面向校招】全力备战2023Golang实习与校招

欢迎群共同进步&＃xff1a;
QQ群&＃xff1a;1007576722